package com.cy.study.controller;

import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.cy.study.common.Response;
import com.cy.study.req.DeleteFileReq;
import com.cy.study.req.FilePageReq;
import com.cy.study.req.UserPageFileReq;
import com.cy.study.res.FileDetailRes;
import com.cy.study.res.FilePageRes;
import com.cy.study.res.UserFileDetailRes;
import com.cy.study.res.UserPageFileRes;
import com.cy.study.service.FileService;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.bind.annotation.*;

import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.net.URLEncoder;

import static com.cy.study.common.CommonConstant.FILE_PATH;

/**
 * 资源控制器
 */
@RequiredArgsConstructor
@RequestMapping("/file")
@RestController
@CrossOrigin
@Slf4j
public class FileController {

    private final FileService fileService;

    /**
     * 管理端文件列表
     * @param req
     * @return
     */
    @PostMapping("/page")
    public Response<Page<FilePageRes>> filePage(@RequestBody FilePageReq req){
        Page<FilePageRes> res = fileService.filePage(req);
        return Response.success(res);
    }

    /**
     * 管理端文件详情
     * @param fileId
     * @return
     */
    @GetMapping("/detail/{fileId}")
    public Response<FileDetailRes> fileDetail(@PathVariable Integer fileId){
        FileDetailRes res = fileService.fileDetail(fileId);
        return Response.success(res);
    }

    /**
     * 管理端删除文件
     * @param req
     * @return
     */
    @PostMapping("/delete")
    public Response deleteFile(@RequestBody DeleteFileReq req){
        fileService.deleteFile(req);
        return Response.success();
    }

    //-----------------------------用户端


    /**
     * 用户端资源列表
     * @param req
     * @return
     */
    @PostMapping("/userPageFile")
    public Response<Page<UserPageFileRes>> userPageFile(@RequestBody UserPageFileReq req){
        Page<UserPageFileRes> res = fileService.userPageFile(req);
        return Response.success(res);
    }


    /**
     * 用户端资源详情
     * @param fileId
     * @return
     */
    @GetMapping("/userFileDetail/{fileId}")
    public Response<UserFileDetailRes> userFileDetail(@PathVariable Integer fileId){
        UserFileDetailRes res = fileService.userFileDetail(fileId);
        return Response.success(res);
    }

    /**
     * 文件下载
     * @param path
     * @param response
     */
    @GetMapping("/download")
    public void download(String path, HttpServletResponse response) {
        try {
            String preStr = FILE_PATH;
            // path是指想要下载的文件的路径
            File file = new File(preStr+path);
            log.info(file.getPath());
            // 获取文件名
            String filename = file.getName();
            // 获取文件后缀名
            String ext = filename.substring(filename.lastIndexOf(".") + 1).toLowerCase();
            log.info("文件后缀名：" + ext);
            // 将文件写入输入流
            FileInputStream fileInputStream = new FileInputStream(file);
            InputStream fis = new BufferedInputStream(fileInputStream);
            byte[] buffer = new byte[fis.available()];
            fis.read(buffer);
            fis.close();
            // 清空response
            response.reset();
            // 设置response的Header
            response.setCharacterEncoding("UTF-8");
            //Content-Disposition的作用：告知浏览器以何种方式显示响应返回的文件，用浏览器打开还是以附件的形式下载到本地保存
            //attachment表示以附件方式下载 inline表示在线打开 "Content-Disposition: inline; filename=文件名.mp3"
            // filename表示文件的默认名称，因为网络传输只支持URL编码的相关支付，因此需要将文件名URL编码后进行传输,前端收到后需要反编码才能获取到真正的名称
            response.addHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode(filename, "UTF-8"));
            // 告知浏览器文件的大小
            response.addHeader("Content-Length", "" + file.length());
            OutputStream outputStream = new BufferedOutputStream(response.getOutputStream());
            response.setContentType("application/octet-stream");
            outputStream.write(buffer);
            outputStream.flush();
            fileService.downloadCount(path);
        } catch (IOException ex) {
            ex.printStackTrace();
        }
    }

}
